NumPy arange()函数的用法

在 NumPy 中，可以使用 arange() 来创建元素值在“指定范围”内的一维数组。需要特别注意的是，arange() 只能创建一维数组，而不能创建多维数组。

使用 arange() 函数的语法格式如下：

np.arange(start, end, step)

start 是开始值，end 是结束值，step 是步长。np.arange() 的取值范围为 [start, end)，也就是包含开始值，但是不包含结束值：

• 当 np.arange() 有一个参数时，表示只有结束值，此时开始值是 0。
• 当 np.arange() 有两个参数时，表示只有开始值和结束值，此时开始值是 start。
• 当 np.arange() 有三个参数时，表示有开始值、结束值和步长。所谓步长，指的是元素递增的量。

【实例 1】一个参数。

import numpy as np
arr = np.arange(10)
print(arr)

输出结果如下：

[0 1 2 3 4 5 6 7 8 9]

np.arange(10) 表示结束值为 10，也就是取值范围为 [0, 10)。从结果可以看出，结束值是没有包含进去的。读者记住这一点就可以了：不管 np.arange() 的参数是多少个，结束值都不会被包含进去。

对于这个例子来说，下面 3 种方式是等价的，读者可以思考一下为什么：

# 方式1
np.arange(10)
# 方式2
np.arange(0, 10)
# 方式3
np.arange(0, 10, 1)

【实例 2】两个参数。

import numpy as np
arr = np.arange(5, 10)
print(arr)

输出结果如下：

[5 6 7 8 9]

np.arange(5, 10) 表示开始值为 5、结束值为 10，也就是取值范围为 [5, 10)。对于这个例子来说，下面两种方式是等价的：

# 方式1
np.arange(5, 10)
# 方式2
np.arange(5, 10, 1)

【举例】3个参数。

import numpy as np
arr = np.arange(10, 30, 3)
print(arr)

输出结果如下：

[10 13 16 19 22 25 28]

np.arange(10, 30, 3) 的取值范围为 [10, 30)，步长为 3。np.arange() 不仅可以用于创建元素的类型是整型的数组，还可以用于创建元素的类型是浮点型的数组。

【实例 4】元素是浮点数

import numpy as np
arr = np.arange(1.5, 10.5, 2)
print(arr)

输出结果如下：

[1.5 3.5 5.5 7.5 9.5]

np.arange(1.5, 10.5, 2) 的取值范围为 [1.5, 10.5)，步长为 2。

【实例 5】步长是浮点数。

import numpy as np
arr = np.arange(1, 10, 1.5)
print(arr)

输出结果如下：

[1.  2.5 4.  5.5 7.  8.5]

np.arange(1, 10, 1.5) 的取值范围为 [1, 10)，步长为 1.5。np.arange() 和 range() 非常相似，但是它们之间也有一定的区别：range() 的步长只能是整数，而 np.arange() 的步长可以是任意数。

此外，arange() 只能创建一维数组。如果想要创建多维数组，可以结合 reshape() 函数来实现。